from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer


'''
This is an example showing how to train a chat bot using the
ChatterBot ListTrainer.
'''

chatbot = ChatBot('Hyundai Car Bot')

# Start by training our bot with the ChatterBot corpus data
trainer = ListTrainer(chatbot)

trainer.train([
    '我的车出故障了吗？',
    '2024-05-05日 12点35分发生轮胎问题。 除此之外，没有故障信号。.',
    '无法远程启动?',
    '现代，5月1日开通，目前服务正常。 车辆的网络状态可能不太好。 请移动到网络状态好的地区使用',
    '故障通知',
    '故障通知 [2024-05-05日 12点35分 轮胎压力问题] 发生 要移动到附近的维修站接受维修. [直接去附近的维修站 400-200-1234]'
])

'''
# You can train with a second list of data to add response variations
trainer.train([
    '我的车不启动怎么办?',
    '我看以后先看看服务开通.',
    '手机网路看看.',
    '再次试试一下'
    '故障通知',
    '故障通知
[2024-05-05日 12点35分 轮胎压力问题] 发生 要移动到附近的维修站接受维修。
[直接去附近的维修站 400-200-1234]'
])
'''
#trainer.train("chatterbot.corpus.hyundai_carbot_chinese")
#trainer.train("chatterbot.corpus.english1")

while True:
    request=input('Question:')
# Get a response to the input text 'I would like to book a flight.'
    response = chatbot.get_response(request)
    print('Hyundai CAR Bot: ',response)
    #print('CAR Bot--> responseCarBot: ',response)